package com.ljt.openapi.demo.util;

import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpDelete;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.message.BasicNameValuePair;

public class HttpUtils {

  /**
   * get
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @return
   * @throws Exception
   */
  public static HttpResponse doGet(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys) throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpGet request = new HttpGet(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    return httpClient.execute(request);
  }

  /**
   * post form
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @param bodys
   * @return
   * @throws Exception
   */
  public static HttpResponse doPost(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys, Map<String, String> bodys)
      throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpPost request = new HttpPost(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    if (bodys != null) {
      List<NameValuePair> nameValuePairList = new ArrayList<NameValuePair>();

      for (String key : bodys.keySet()) {
        nameValuePairList.add(new BasicNameValuePair(key, bodys.get(key)));
      }
      UrlEncodedFormEntity formEntity = new UrlEncodedFormEntity(nameValuePairList, "utf-8");
      formEntity.setContentType("application/x-www-form-urlencoded; charset=UTF-8");
      request.setEntity(formEntity);
    }

    return httpClient.execute(request);
  }

  /**
   * Post String
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @param body
   * @return
   * @throws Exception
   */
  public static HttpResponse doPost(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys, String body) throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpPost request = new HttpPost(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    if (StringUtils.isNotBlank(body)) {
      request.setEntity(new StringEntity(body, "utf-8"));
    }

    return httpClient.execute(request);
  }

  /**
   * Post stream
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @param body
   * @return
   * @throws Exception
   */
  public static HttpResponse doPost(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys, byte[] body) throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpPost request = new HttpPost(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    if (body != null) {
      request.setEntity(new ByteArrayEntity(body));
    }

    return httpClient.execute(request);
  }

  /**
   * Put String
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @param body
   * @return
   * @throws Exception
   */
  public static HttpResponse doPut(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys, String body) throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpPut request = new HttpPut(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    if (StringUtils.isNotBlank(body)) {
      request.setEntity(new StringEntity(body, "utf-8"));
    }

    return httpClient.execute(request);
  }

  /**
   * Put stream
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @param body
   * @return
   * @throws Exception
   */
  public static HttpResponse doPut(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys, byte[] body) throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpPut request = new HttpPut(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    if (body != null) {
      request.setEntity(new ByteArrayEntity(body));
    }

    return httpClient.execute(request);
  }

  /**
   * Delete
   * @param host
   * @param path
   * @param method
   * @param headers
   * @param querys
   * @return
   * @throws Exception
   */
  public static HttpResponse doDelete(String host, String path, String method,
      Map<String, String> headers, Map<String, String> querys) throws Exception {
    HttpClient httpClient = wrapClient(host);

    HttpDelete request = new HttpDelete(buildUrl(host, path, querys));
    for (Map.Entry<String, String> e : headers.entrySet()) {
      request.addHeader(e.getKey(), e.getValue());
    }

    return httpClient.execute(request);
  }

  private static String buildUrl(String host, String path, Map<String, String> querys)
      throws UnsupportedEncodingException {
    StringBuilder sbUrl = new StringBuilder();
    sbUrl.append(host);
    if (!StringUtils.isBlank(path)) {
      sbUrl.append(path);
    }
    if (null != querys) {
      StringBuilder sbQuery = new StringBuilder();
      for (Map.Entry<String, String> query : querys.entrySet()) {
        if (0 < sbQuery.length()) {
          sbQuery.append("&");
        }
        if (StringUtils.isBlank(query.getKey()) && !StringUtils.isBlank(query.getValue())) {
          sbQuery.append(query.getValue());
        }
        if (!StringUtils.isBlank(query.getKey())) {
          sbQuery.append(query.getKey());
          if (!StringUtils.isBlank(query.getValue())) {
            sbQuery.append("=");
            sbQuery.append(URLEncoder.encode(query.getValue(), "utf-8"));
          }
        }
      }
      if (0 < sbQuery.length()) {
        sbUrl.append("?").append(sbQuery);
      }
    }

    return sbUrl.toString();
  }

  private static HttpClient wrapClient(String host) {
    HttpClientBuilder httpBuilder = HttpClientBuilder.create();
    //HttpClient httpClient = httpBuilder.build();
//    if (host.startsWith("https://")) {
//      sslClient(httpBuilder);
//    }

    return httpBuilder.build();
  }

//  private static void sslClient(HttpClientBuilder httpBuilder) {
//    try {
//      SSLContext ctx = SSLContext.getInstance("TLS");
//      X509TrustManager tm = new X509TrustManager() {
//        public X509Certificate[] getAcceptedIssuers() {
//          return null;
//        }
//
//        public void checkClientTrusted(X509Certificate[] xcs, String str) {
//
//        }
//
//        public void checkServerTrusted(X509Certificate[] xcs, String str) {
//
//        }
//      };
//      ctx.init(null, new TrustManager[] {tm}, null);
//      SSLConnectionSocketFactory ssf = new SSLConnectionSocketFactory(ctx);
//      
//      //SSLSocketFactory ssf = new SSLSocketFactory(ctx);
//      //ssf.setHostnameVerifier(SSLSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER);
//      HttpClientConnectionManager ccm = new BasicHttpClientConnectionManager();
//      org.apache.http.config.Registry registry = ccm.;
//      registry.register(new Scheme("https", 443, ssf));
//    } catch (KeyManagementException ex) {
//      throw new RuntimeException(ex);
//    } catch (NoSuchAlgorithmException ex) {
//      throw new RuntimeException(ex);
//    }
//  }
}
